package classifiers;

import mulan.classifier.MultiLabelOutput;

public class PredictionForCluster {
	
	private int cluster_idx;
	private double multiplier;
	
	private MultiLabelOutput prediction;

	public int getCluster_idx() {
		return cluster_idx;
	}

	public void setCluster_idx(int cluster_idx) {
		this.cluster_idx = cluster_idx;
	}

	public MultiLabelOutput getPrediction() {
		return prediction;
	}

	public void setPrediction(MultiLabelOutput prediction) {
		this.prediction = prediction;
	}

	public PredictionForCluster(int cluster_idx, double multiplier,MultiLabelOutput prediction) {
		this.cluster_idx = cluster_idx;
		this.prediction = prediction;
		this.multiplier = multiplier;
	}

	public double getMultiplier() {
		return multiplier;
	}

	public void setMultiplier(double multiplier) {
		this.multiplier = multiplier;
	}
	
	public double getConfidence(int i) {
		return prediction.getConfidences()[i]*multiplier;
	}
	

}
